package com.lin.generator;

import cn.hutool.core.io.FileUtil;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.Writer;

public class DynamicFileGenerator {
    public static void DynamicFileGenerator(String inputDynamicGeneratorFile,String outputDynamicGeneratorFile,Object model) throws IOException, TemplateException {
        doGenerator(inputDynamicGeneratorFile,outputDynamicGeneratorFile, model);
    }

    public static void doGenerator(String inputPath,String outputPath,Object dataModel) throws IOException, TemplateException {
        // 1：创建配置对象
        // new 出 Configuration 对象，参数为 FreeMarker 版本号
        Configuration configuration = new Configuration(Configuration.VERSION_2_3_32);
        // 指定模板文件所在的路径
        configuration.setDirectoryForTemplateLoading(new File(inputPath).getParentFile());
        // 设置模板文件使用的字符集
        configuration.setDefaultEncoding("utf-8");
        // 创建模板对象，加载指定模板
        Template template = configuration.getTemplate(new File(inputPath).getName());
        // 2：指定生成的文件
        // 如果文件不存在先生成一个文件
        if(!FileUtil.exist(outputPath)){
            FileUtil.touch(outputPath);
        }
        Writer out = new FileWriter(outputPath);
        // 3:生成文件
        template.process(dataModel, out);
        // 4:关闭流
        out.close();
    }
}
